# 写一个 lambda 表达式来创建函数，
# fx = lambda n: ....
# 此函数返回n的平方加1能否被5整除，如果能整除，返回True,否则返回False
# print(fx(3))  # True
# print(fx(4))  # False

fx = lambda n: (n ** 2 + 1) % 5 == 0

print(fx(3))  # True
print(fx(4))  # False